package com.example.demo.config;

import io.milvus.v2.client.ConnectConfig;
import io.milvus.v2.client.MilvusClientV2;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

@Component
public class MilvusConfig {

    @Value("${milvus.host}")
    String host;

    @Value("${milvus.port}")
    String port;

    @Bean
    public MilvusClientV2 getClient() {
        String uri = String.format("http://%s:%s", host, port);
        ConnectConfig connectConfig = ConnectConfig.builder().uri(uri).build();

        return new MilvusClientV2(connectConfig);
    }
}
